import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;

import java.util.TreeSet;
import java.util.stream.Stream;

/**
 * 存在重复元素III
 * 题目：在整数数组 nums 中，是否存在两个下标 i 和 j，使得nums [i] 和nums [j]的差的绝对值小于等于 t ，且满足 i 和 j 的差的绝对值也小于等于 ķ 。
 * 如果存在则返回 true，不存在返回 false。
 * <p>
 * 示例 1:
 * 输入: nums = [1,2,3,1], k = 3, t = 0
 * 输出: true
 * <p>
 * 示例 2:
 * 输入: nums = [1,0,1,1], k = 1, t = 2
 * 输出: true
 * <p>
 * 示例 3:
 * 输入: nums = [1,5,9,1,5,9], k = 2, t = 3
 * 输出: false
 * <p>
 * 来源：力扣（LeetCode-220）
 * 链接：https://leetcode-cn.com/problems/contains-duplicate-ii
 *
 * @author godfrey
 * @since 2020-12-03
 */
@DisplayName("存在重复元素III")
public class ContainsNearbyAlmostDuplicate extends BaseTest {

    static Stream<Arguments> testArguments() {
        return Stream.of(
                //Arguments.arguments(new int[]{1, 2, 3, 1}, 3, 0),
                //Arguments.arguments(new int[]{1, 0, 1, 1}, 1, 2),
                //Arguments.arguments(new int[]{1, 5, 9, 1, 5, 9}, 2, 3),
                Arguments.arguments(new int[]{2147483640, 2147483641}, 1, 100)
        );
    }

    @DisplayName("BST二叉搜索树+滑动窗口-时间复杂度O(nlogk)，空间复杂度O(k)")
    @ParameterizedTest
    @MethodSource("testArguments")
    void containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if (k < 1 || t < 0) {
            System.out.println(false);
            return;
        }
        final TreeSet<Integer> set = new TreeSet<>();
        for (int i = 0; i < nums.length; ++i) {
            final int x = nums[i];
            //返回集合小于或等于x的最大元素，最大下限
            final Integer floor = set.floor(x);
            //返回集合大于或等于x的最小元素，最小上限
            final Integer ceiling = set.ceiling(x);
            if ((floor != null && x <= floor + t) || (ceiling != null && x >= ceiling - t)) {
                System.out.println(true);
                return;
            }
            set.add(x);
            if (i >= k) {
                set.remove(nums[i - k]);
            }
        }
        System.out.println(false);
    }
}
